51f360b1e2833f78b8515bebef5fc3d7e384e19b,h2o-algos/src/main/java/hex/deeplearning/DeepLearning.java,DeepLearning,trainModel,#DeepLearningModel#,961
Before Change
float[] trainSamplingFactors;
if (mp.classification && mp.balance_classes) {
trainSamplingFactors = new float[train.lastVec().domain().length]; //leave initialized to 0 -> will be filled up below
train = updateFrame(train, sampleFrameStratified(
train, train.lastVec(), trainSamplingFactors, (long)(mp.max_after_balance_size*train.numRows()), mp.seed, true, false));
model.setModelClassDistribution(new MRUtils.ClassDist(train.lastVec()).doAll(train.lastVec()).rel_dist());
}
model.training_rows = train.numRows();
After Change
float[] trainSamplingFactors;
if (mp.classification && mp.balance_classes) {
trainSamplingFactors = new float[train.lastVec().domain().length]; //leave initialized to 0 -> will be filled up below
train = sampleFrameStratified(
train, train.lastVec(), trainSamplingFactors, (long)(mp.max_after_balance_size*train.numRows()), mp.seed, true, false);
model.setModelClassDistribution(new MRUtils.ClassDist(train.lastVec()).doAll(train.lastVec()).rel_dist());
}
model.training_rows = train.numRows();